Update app.py

app.py

@@ -15,6 +15,7 @@ import requests
 import json
 import re
 from datetime import datetime, timedelta
+from pathlib import Path
 
 # Load environment variables
 load_dotenv()
@@ -433,30 +434,59 @@ def perform_stack_search(query, tag, sort_by):
     except Exception as e:
         return f"Error searching Stack Overflow: {str(e)}"
 
-# Gradio interface
+# Modify the file input and processing section
+def process_code_file(file_obj):
+    """Process uploaded code files"""
+    if file_obj is None:
+        return None, "No file uploaded", {}
+    try:
+        content = file_obj.read().decode('utf-8')
+        file_extension = Path(file_obj.name).suffix.lower()
+        language = detect_language(file_extension)
+        
+        # Calculate metrics
+        metrics = calculate_complexity_metrics(content, language)
+        
+        return str(uuid.uuid4()), f"✅ Successfully analyzed {file_obj.name}", metrics
+    except Exception as e:
+        return None, f"Error processing file: {str(e)}", {}
+
+# Update the Gradio interface
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     current_session_id = gr.State(None)
-    pdf_state = gr.State({"page_images": [], "total_pages": 0, "total_words": 0})
+    code_state = gr.State({})
+    
     gr.HTML("""
     <div class="header">
         <div class="header-title">Tech-Vision</div>
-        <div class="header-subtitle">Analyze technical documents with Groq's LLM API.</div>
+        <div class="header-subtitle">Advanced Code Analysis System</div>
     </div>
     """)
+    
     with gr.Row(elem_classes="container"):
         with gr.Column(scale=1, min_width=300):
-            pdf_file = gr.File(label="Upload PDF Document", file_types=[".pdf"], type="binary")
-            upload_button = gr.Button("Process PDF", variant="primary")
-            pdf_status = gr.Markdown("No PDF uploaded yet")
+            file_input = gr.File(
+                label="Upload Code File",
+                file_types=[
+                    ".py", ".js", ".java", ".cpp", ".c", ".cs", ".php",
+                    ".rb", ".go", ".rs", ".swift", ".kt", ".ts", ".html",
+                    ".css", ".sql", ".r", ".m", ".h", ".hpp", ".jsx",
+                    ".tsx", ".vue", ".scala", ".pl", ".sh", ".bash",
+                    ".ps1", ".yaml", ".yml", ".json", ".xml", ".toml", ".ini"
+                ],
+                type="binary"
+            )
+            upload_button = gr.Button("Analyze Code", variant="primary")
+            file_status = gr.Markdown("No file uploaded yet")
             model_dropdown = gr.Dropdown(
                 choices=["llama3-70b-8192", "llama3-8b-8192", "mixtral-8x7b-32768", "gemma-7b-it"],
                 value="llama3-70b-8192",
                 label="Select Groq Model"
             )
             
-            # Tech Tools Section
+            # Developer Tools Section
             gr.Markdown("### Developer Tools", elem_classes="tool-title")
-            with gr.Box(elem_classes="tool-container"):
+            with gr.Group(elem_classes="tool-container"):  # Replace Box with Group
                 with gr.Tabs():
                     with gr.TabItem("GitHub Search"):
                         repo_query = gr.Textbox(label="Search Query", placeholder="Enter keywords to search for repositories")
@@ -503,11 +533,10 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                 
         with gr.Column(scale=2, min_width=600):
             with gr.Tabs():
-                with gr.TabItem("PDF Viewer"):
-                    with gr.Column(elem_classes="pdf-viewer-container"):
-                        page_slider = gr.Slider(minimum=1, maximum=1, step=1, label="Page Number", value=1)
-                        pdf_image = gr.Image(label="PDF Page", type="pil", elem_classes="pdf-viewer-image")
-                        stats_display = gr.Markdown("No PDF uploaded yet", elem_classes="stats-box")
+                with gr.TabItem("Code Analysis"):
+                    with gr.Column(elem_classes="code-viewer-container"):
+                        code_metrics = gr.Markdown("No code analyzed yet", elem_classes="stats-box")
+                        code_recommendations = gr.Markdown("", elem_classes="recommendations-box")
                 
                 with gr.TabItem("GitHub Results"):
                     repo_results = gr.Markdown("Search for repositories to see results here")
@@ -526,15 +555,28 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                 send_btn = gr.Button("Send", scale=1)
             clear_btn = gr.Button("Clear Conversation")
     
-    # Event Handlers
+    # Update event handlers
     upload_button.click(
-        process_pdf,
-        inputs=[pdf_file],
-        outputs=[current_session_id, pdf_status, pdf_state]
+        process_code_file,
+        inputs=[file_input],
+        outputs=[current_session_id, file_status, code_state]
+    ).then(
+        lambda state: (
+            f"### Code Analysis Results\n\n"
+            f"**Language:** {state.get('language', 'Unknown')}\n"
+            f"**Total Lines:** {state.get('total_lines', 0)}\n"
+            f"**Code Lines:** {state.get('code_lines', 0)}\n"
+            f"**Comment Lines:** {state.get('comments', 0)}\n"
+            f"**Functions:** {state.get('functions', 0)}\n"
+            f"**Classes:** {state.get('classes', 0)}\n"
+            f"**Complexity Score:** {state.get('cyclomatic_complexity', 0)}\n"
+        ),
+        inputs=[code_state],
+        outputs=[code_metrics]
     ).then(
-        update_pdf_viewer,
-        inputs=[pdf_state],
-        outputs=[page_slider, pdf_image, stats_display]
+        lambda state: generate_recommendations(state),
+        inputs=[code_state],
+        outputs=[code_recommendations]
     )
     
     msg.submit(
@@ -589,4 +631,85 @@ gr.HTML("""
 
 # Launch the app
 if __name__ == "__main__":
-    demo.launch() 
+    demo.launch() 
+
+# Add new helper functions
+def detect_language(extension):
+    """Map file extensions to programming languages"""
+    extension_map = {
+        ".py": "Python",
+        ".js": "JavaScript",
+        ".java": "Java",
+        ".cpp": "C++",
+        ".c": "C",
+        ".cs": "C#",
+        ".php": "PHP",
+        ".rb": "Ruby",
+        ".go": "Go",
+        ".rs": "Rust",
+        ".swift": "Swift",
+        ".kt": "Kotlin",
+        ".ts": "TypeScript",
+        ".html": "HTML",
+        ".css": "CSS",
+        ".sql": "SQL",
+        ".r": "R",
+        ".scala": "Scala",
+        ".pl": "Perl",
+        ".sh": "Shell",
+        ".yaml": "YAML",
+        ".yml": "YAML",
+        ".json": "JSON",
+        ".xml": "XML",
+        ".jsx": "React JSX",
+        ".tsx": "React TSX",
+        ".vue": "Vue",
+    }
+    return extension_map.get(extension.lower(), "Unknown")
+
+def calculate_complexity_metrics(content, language):
+    """Calculate code complexity metrics"""
+    lines = content.split('\n')
+    total_lines = len(lines)
+    blank_lines = len([line for line in lines if not line.strip()])
+    code_lines = total_lines - blank_lines
+    
+    # Get language patterns
+    patterns = LANGUAGE_PATTERNS.get(language.lower(), LANGUAGE_PATTERNS.get("python"))
+    
+    # Calculate metrics using patterns
+    metrics = {
+        "language": language,
+        "total_lines": total_lines,
+        "code_lines": code_lines,
+        "blank_lines": blank_lines,
+        "functions": len(re.findall(patterns["function"], content, re.MULTILINE)) if patterns else 0,
+        "classes": len(re.findall(patterns["class"], content, re.MULTILINE)) if patterns else 0,
+        "imports": len(re.findall(patterns["import"], content, re.MULTILINE)) if patterns else 0,
+        "comments": len(re.findall(patterns["comment"], content, re.MULTILINE)) if patterns else 0,
+        "conditionals": len(re.findall(patterns["conditional"], content, re.MULTILINE)) if patterns else 0,
+        "loops": len(re.findall(patterns["loop"], content, re.MULTILINE)) if patterns else 0,
+    }
+    
+    # Calculate cyclomatic complexity
+    metrics["cyclomatic_complexity"] = 1 + metrics["conditionals"] + metrics["loops"]
+    
+    return metrics
+
+def generate_recommendations(metrics):
+    """Generate code quality recommendations based on metrics"""
+    recommendations = []
+    
+    if metrics.get("cyclomatic_complexity", 0) > 10:
+        recommendations.append("🔄 High cyclomatic complexity detected. Consider breaking down complex functions.")
+    
+    if metrics.get("code_lines", 0) > 300:
+        recommendations.append("📏 File is quite large. Consider splitting it into multiple modules.")
+    
+    if metrics.get("functions", 0) > 10:
+        recommendations.append("🔧 Large number of functions. Consider grouping related functions into classes.")
+    
+    if metrics.get("comments", 0) / max(metrics.get("code_lines", 1), 1) < 0.1:
+        recommendations.append("📝 Low comment ratio. Consider adding more documentation.")
+    
+    return "### Recommendations\n\n" + "\n\n".join(recommendations) if recommendations else ""