Update app.py

app.py

@@ -4,6 +4,8 @@ import openai
 import faiss
 import os
 import numpy as np
+from sklearn.feature_extraction.text import TfidfVectorizer
+from sklearn.metrics.pairwise import cosine_similarity
 from io import StringIO
 
 # Function to extract text from a PDF file
@@ -118,10 +120,43 @@ if openai_api_key:
 
                 # Display the response in Streamlit (Intelligent Reply)
                 st.write("### Intelligent Reply:")
-                st.text_area("Response:", response_content, height=300)
-
-                # Copy button
-                st.button("Copy Response", on_click=lambda: st.text_area("Copy the response", response_content, height=300))
+                st.write(response_content)
+
+    elif mode == "Code Generator":
+        st.header("Code Generator")
+
+        # Code generation prompt input
+        code_prompt = st.text_area("Describe the code you want to generate:", 
+                                   "e.g., Write a Python program that generates Fibonacci numbers.")
+        
+        if st.button("Generate Code"):
+            if code_prompt:
+                with st.spinner("Generating code..."):
+                    # Generate code using GPT-4
+                    generated_code = generate_code_from_prompt(code_prompt)
+                    
+                    # Clean the generated code to ensure only code is saved (removing comments or additional text)
+                    clean_code = "\n".join([line for line in generated_code.splitlines() if not line.strip().startswith("#")])
+
+                    # Save the clean code to a file
+                    save_code_to_file(clean_code)
+
+                    # Display the generated code
+                    st.write("### Generated Code:")
+                    st.code(clean_code, language="python")
+
+                    # Provide a download link for the generated code
+                    with open("generated_code.txt", "w") as f:
+                        f.write(clean_code)
+
+                    st.download_button(
+                        label="Download Generated Code",
+                        data=open("generated_code.txt", "rb").read(),
+                        file_name="generated_code.txt",
+                        mime="text/plain"
+                    )
+            else:
+                st.error("Please provide a prompt to generate the code.")
 
     elif mode == "AI Chatbot Tutor":
         st.header("AI Chatbot Tutor")
@@ -145,10 +180,6 @@ if openai_api_key:
                 bot_response = chat_with_bot(user_query)
                 st.write(f"### AI Response: {bot_response}")
 
-                # Copy button
-                st.text_area("Copy the response", bot_response, height=300)
-                st.button("Copy Response", on_click=lambda: st.text_area("Copy the response", bot_response, height=300))
-
     elif mode == "AI Study Notes & Summaries":
         st.header("AI Study Notes & Summaries")
 
@@ -169,7 +200,4 @@ if openai_api_key:
 
             # Display the summary
             st.write("### AI-Generated Summary:")
-            st.text_area("Summary:", summary, height=300)
-
-            # Copy button
-            st.button("Copy Summary", on_click=lambda: st.text_area("Copy the summary", summary, height=300))
+            st.write(summary)