Update app.py

app.py

@@ -38,6 +38,17 @@ def generate_filename(prompt, file_type):
     safe_prompt = "".join(x for x in prompt if x.isalnum())[:28]
     return f"{safe_date_time}_{safe_prompt}.{file_type}"
 
+def create_file(filename, prompt, response):
+    if filename.endswith(".txt"):
+        with open(filename, 'w') as file:
+            file.write(f"Prompt:\n{prompt}\nResponse:\n{response}")
+    elif filename.endswith(".htm"):
+        with open(filename, 'w') as file:
+            file.write(f"<h1>Prompt:</h1> <p>{prompt}</p> <h1>Response:</h1> <p>{response}</p>")
+    elif filename.endswith(".md"):
+        with open(filename, 'w') as file:
+            file.write(f"# Prompt:\n{prompt}\n# Response:\n{response}")
+
 def truncate_document(document, length):
     return document[:length]