Update app.py

app.py

@@ -1,8 +1,23 @@
 import streamlit as st
+from PIL import Image
+import tempfile
+import os
+import numpy as np
+import cv2
+import torch
 import difflib
+import re
 import requests
 import datetime
 import streamlit.components.v1 as components
+import io
+
+from diffusers import DiffusionPipeline
+from moviepy.editor import ImageSequenceClip
+
+import insightface
+from insightface.app import FaceAnalysis
+from insightface.utils import face_align
 
 # --- CONFIG ---
 # Place your API keys here
@@ -103,6 +118,14 @@ def get_explanation_prompt(code, programming_language, skill_level, user_role, e
         f"Explain this {programming_language} code for a {skill_level} {user_role}.{lang_instruction}\n{code}"
     )
 
+# --- SESSION STATE FOR CHAT HISTORY ---
+if "workflow_history" not in st.session_state:
+    st.session_state.workflow_history = []
+if "semantic_history" not in st.session_state:
+    st.session_state.semantic_history = []
+if "comment_history" not in st.session_state:
+    st.session_state.comment_history = []
+
 # --- STREAMLIT APP ---
 st.set_page_config(page_title="Code Workflows", layout="wide")
 st.title("Code Genie")
@@ -113,7 +136,7 @@ page = st.sidebar.radio("Navigate", ["Home", "Code Workflows", "Semantic Search"
 if page == "Home":
     st.header("Welcome to the Code Genie!")
     st.markdown("""
-    - **Full Code Workflow:** Complete code analysis pipeline with explanation, refactoring, review, and testing
+    - **Full Code Workflow:** Complete code analysis pipeline with explanation, refactoring, review, and testing (powered by Groq/Blackbox)
     - **Semantic Search:** Ask natural language questions about your code and get intelligent answers
     - **Code Comment Generator:** Helps you add helpful comments to your code for better readability
     """)
@@ -170,6 +193,32 @@ elif page == "Code Workflows":
                 for t in timeline:
                     report += f"## {t['step']}\n{t['output']}\n\n---\n\n"
                 st.download_button("Download Report", report, file_name="ai_workflow_report.txt")
+                # Save to chat history
+                st.session_state.workflow_history.append({
+                    "timestamp": str(datetime.datetime.now()),
+                    "user_code": code_input,
+                    "params": {
+                        "language": programming_language,
+                        "skill": skill_level,
+                        "role": user_role,
+                        "explanation_language": explanation_language
+                    },
+                    "timeline": timeline,
+                    "refactored_code": refactored_code
+                })
+    # Show chat history for workflows
+    st.markdown("### Workflow Chat History")
+    if st.button("Clear Workflow History"):
+        st.session_state.workflow_history = []
+    for entry in reversed(st.session_state.workflow_history):
+        st.markdown(f"**[{entry['timestamp']}]**")
+        st.code(entry["user_code"], language=entry["params"]["language"].lower())
+        for t in entry["timeline"]:
+            st.subheader(t["step"])
+            st.write(t["output"])
+        st.subheader("Code Diff (Original vs Refactored)")
+        st.code(get_inline_diff(entry["user_code"], entry["refactored_code"]), language=entry["params"]["language"].lower())
+        st.markdown("---")
 
 elif page == "Semantic Search":
     st.header("Semantic Search")
@@ -191,7 +240,7 @@ elif page == "Semantic Search":
     st.caption("Example questions:")
     st.write(", ".join(EXAMPLE_QUESTIONS))
 
-    # Single input field for question (typed only)
+    # Only text input for question
     question = st.text_input("Ask a question about your code", key="sem_question")
 
     # Run Semantic Search button
@@ -206,6 +255,29 @@ elif page == "Semantic Search":
                 answer = call_groq_api(prompt)
                 st.success("Answer:")
                 st.write(answer)
+                # Save to chat history
+                st.session_state.semantic_history.append({
+                    "timestamp": str(datetime.datetime.now()),
+                    "user_code": code_input,
+                    "question": question,
+                    "params": {
+                        "language": programming_language,
+                        "skill": skill_level,
+                        "role": user_role,
+                        "explanation_language": explanation_language
+                    },
+                    "answer": answer
+                })
+    # Show chat history for semantic search
+    st.markdown("### Semantic Search Chat History")
+    if st.button("Clear Semantic History"):
+        st.session_state.semantic_history = []
+    for entry in reversed(st.session_state.semantic_history):
+        st.markdown(f"**[{entry['timestamp']}]**")
+        st.code(entry["user_code"], language=entry["params"]["language"].lower())
+        st.markdown(f"**Q:** {entry['question']}")
+        st.markdown(f"**A:** {entry['answer']}")
+        st.markdown("---")
 
 elif page == "Code Comment Generator":
     st.header("Code Comment Generator")
@@ -234,4 +306,40 @@ elif page == "Code Comment Generator":
                 ])
                 st.success("Commented code generated!")
                 st.code(commented_code, language=programming_language.lower())
-                st.download_button("Download Commented Code", commented_code, file_name="commented_code.txt") 
+                st.download_button("Download Commented Code", commented_code, file_name="commented_code.txt")
+                # Save to chat history
+                st.session_state.comment_history.append({
+                    "timestamp": str(datetime.datetime.now()),
+                    "user_code": code_input,
+                    "params": {
+                        "language": programming_language
+                    },
+                    "commented_code": commented_code
+                })
+    # Show chat history for code comments
+    st.markdown("### Code Comment Chat History")
+    if st.button("Clear Comment History"):
+        st.session_state.comment_history = []
+    for entry in reversed(st.session_state.comment_history):
+        st.markdown(f"**[{entry['timestamp']}]**")
+        st.code(entry["user_code"], language=entry["params"]["language"].lower())
+        st.markdown("**Commented Code:**")
+        st.code(entry["commented_code"], language=entry["params"]["language"].lower())
+        st.markdown("---")
+
+st.markdown("---")
+st.write("Powered by AnimateDiff (zeroscope), InsightFace, and moviepy.")
+
+def split_code_into_chunks(code, lang):
+    if lang.lower() == "python":
+        # Corrected regex pattern for Python code splitting
+        pattern = r'(def\s+\w+\(.*?\):|class\s+\w+\(.*?\)?:)'
+        splits = re.split(pattern, code)
+        chunks = []
+        for i in range(1, len(splits), 2):
+            header = splits[i]
+            body = splits[i+1] if (i+1) < len(splits) else ""
+            chunks.append(header + body)
+        return chunks if chunks else [code]
+    else:
+        return [code]