Update app.py

app.py

@@ -12,7 +12,7 @@ load_dotenv()
 openai.api_key = os.getenv('OPENAI_KEY')
 
 def chat_with_model(prompts):
-    model = "gpt-3.5-turbo"  
+    model = "gpt-3.5-turbo"
 
     conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
     conversation.extend([{'role': 'user', 'content': prompt} for prompt in prompts])
@@ -23,17 +23,14 @@ def chat_with_model(prompts):
 def generate_filename(prompt):
     safe_date_time = datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
     safe_prompt = "".join(x for x in prompt if x.isalnum())[:50]
-    return f"{safe_date_time}_{safe_prompt}.md"
+    return f"{safe_date_time}_{safe_prompt}.htm"
 
 def create_file(filename, prompt, response):
     with open(filename, 'w') as file:
-        file.write(f"Prompt: {prompt}\n\nResponse: {response}")
+        file.write(f"<h1>Prompt:</h1> <p>{prompt}</p> <h1>Response:</h1> <p>{response}</p>")
 
 def get_table_download_link(file_path):
-    with open(file_path, 'r') as file:
-        data = file.read()
-    b64 = base64.b64encode(data.encode()).decode()  
-    href = f'<a href="data:file/txt;base64,{b64}" download="{os.path.basename(file_path)}">Download Response File</a>'
+    href = f'<a href="{file_path}" target="_blank">{os.path.basename(file_path)}</a>'
     return href
 
 def main():
@@ -59,8 +56,8 @@ def main():
 
         st.markdown(get_table_download_link(filename), unsafe_allow_html=True)
 
-    md_files = glob.glob("*.md")
-    for file in md_files:
+    htm_files = glob.glob("*.htm")
+    for file in htm_files:
         st.markdown(get_table_download_link(file), unsafe_allow_html=True)
 
 if __name__ == "__main__":