fixes done

Author: Drago-03

app.py

@@ -30,9 +30,9 @@
 
 # Import AI models
 try:
-    from ai_models.github_analyzer import GitHubAnalyzer
-    from ai_models.documentation_generator import DocumentationGenerator
-    from ai_models.rag_pipeline import RAGPipeline
+    from ai_models.github_analyzer import GitHubAnalyzer as GitHubAnalyzerModel
+    from ai_models.documentation_generator import DocumentationGenerator as DocumentationGeneratorModel
+    from ai_models.rag_pipeline import RAGPipeline as RAGPipelineModel
     logger = logging.getLogger(__name__)
     logger.info("AI models imported successfully")
 except ImportError as e:
@@ -48,13 +48,15 @@ def parse_github_url(self, repo_url: str) -> Dict[str, str]:
         def analyze_repository(self, repo_url: str) -> Dict[str, Any]:
             raise Exception("AI models not available - import failed")
 
-    class DocumentationGenerator:
-        def generate_documentation(self, analysis_result: Dict[str, Any]) -> Dict[str, Any]:
+    class DummyDocumentationGenerator:
+        def generate_documentation(self, analysis_result: Dict[str, Any], rag_result: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
             raise Exception("AI models not available - import failed")
 
     class RAGPipeline:
         def process(self, analysis_result: Dict[str, Any]) -> Dict[str, Any]:
             raise Exception("AI models not available - import failed")
+        def process_repository(self, analysis_result: Dict[str, Any]) -> Dict[str, Any]:
+            raise Exception("AI models not available - import failed")
 
 # Load environment variables
 load_dotenv()
@@ -333,7 +335,7 @@ class DocumentationGenerator:
     def __init__(self):
         self.template_path = Path(__file__).parent / 'templates'
 
-    def generate_documentation(self, analysis_result: Dict[str, Any]) -> Dict[str, Any]:
+    def generate_documentation(self, analysis_result: Dict[str, Any], rag_result: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
         """Generate complete documentation package"""
         try:
             repo_info = analysis_result.get('repository_info', {})
@@ -1433,7 +1435,11 @@ def cleanup(self):
 
 # Initialize components with better error handling
 try:
-    github_analyzer = GitHubAnalyzer()
+    # Check if GitHubAnalyzerModel was successfully imported
+    if 'GitHubAnalyzerModel' in locals() or 'GitHubAnalyzerModel' in globals():
+        github_analyzer = GitHubAnalyzerModel()
+    else:
+        github_analyzer = GitHubAnalyzer()
     logger.info("✅ GitHubAnalyzer initialized successfully")
 except Exception as e:
     logger.error(f"❌ Failed to initialize GitHubAnalyzer: {e}")
@@ -1447,19 +1453,19 @@ def analyze_repository(self, repo_url: str) -> Dict[str, Any]:
     github_analyzer = FallbackGitHubAnalyzer()
 
 try:
-    doc_generator = DocumentationGenerator()
+    doc_generator = DocumentationGeneratorModel() if 'DocumentationGeneratorModel' in globals() else DocumentationGenerator()
     logger.info("✅ DocumentationGenerator initialized successfully")
 except Exception as e:
     logger.error(f"❌ Failed to initialize DocumentationGenerator: {e}")
     logger.error(traceback.format_exc())
     # Create a fallback class
     class FallbackDocumentationGenerator:
-        def generate_documentation(self, analysis_result: Dict[str, Any]) -> Dict[str, Any]:
+        def generate_documentation(self, analysis_result: Dict[str, Any], rag_result: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
             raise Exception(f"DocumentationGenerator initialization failed: {e}")
     doc_generator = FallbackDocumentationGenerator()
 
 try:
-    rag_pipeline = RAGPipeline()
+    rag_pipeline = RAGPipelineModel() if 'RAGPipelineModel' in globals() else RAGPipeline()
     logger.info("✅ RAGPipeline initialized successfully")
 except Exception as e:
     logger.error(f"❌ Failed to initialize RAGPipeline: {e}")
@@ -1496,6 +1502,35 @@ def favicon():
     """Handle favicon requests"""
     return '', 204  # No Content
 
+# Handle malformed API URLs with double slashes
+@app.route('/api//<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE'])
+def handle_double_slash_api(path):
+    """Redirect malformed API URLs with double slashes"""
+    logger.warning(f"Received malformed API URL with double slash: /api//{path}")
+    
+    # For analyze endpoint, handle directly instead of redirecting
+    if path == 'analyze':
+        if request.method == 'GET':
+            return jsonify({
+                'message': 'Repository analysis endpoint',
+                'method': 'POST',
+                'required_fields': ['repo_url'],
+                'description': 'Submit a GitHub repository URL to generate AI documentation',
+                'example': {
+                    'repo_url': 'https://github.com/owner/repository'
+                },
+                'note': 'URL was corrected from double slash'
+            })
+        elif request.method == 'POST':
+            # Forward to the actual analyze function
+            return analyze_repository()
+    
+    # For other endpoints, redirect to the corrected URL
+    corrected_url = f"/api/{path}"
+    logger.info(f"Redirecting to: {corrected_url}")
+    from flask import redirect
+    return redirect(corrected_url)
+
 @app.route('/api/health')
 def health_check():
     """Enhanced health check endpoint with detailed system status"""
@@ -1575,6 +1610,7 @@ def health_check():
         }), 500
 
 @app.route('/api/analyze', methods=['GET', 'POST'])
+@app.route('/api/analyze/', methods=['GET', 'POST'])
 def analyze_repository():
     """Handle repository analysis requests"""
     if request.method == 'GET':
@@ -1680,10 +1716,29 @@ def analyze_repository():
                 }
             }), 500
 
-        # Step 2: Generate documentation
+        # Step 2: Process with RAG pipeline
+        try:
+            logger.info("Step 2: Processing with RAG pipeline")
+            rag_result = rag_pipeline.process_repository(analysis_result)
+            logger.info("RAG pipeline processing completed successfully")
+        except Exception as e:
+            logger.error(f"RAG pipeline processing failed: {e}")
+            logger.error(traceback.format_exc())
+            # Continue with empty rag_result as fallback
+            rag_result = {
+                'processing_status': 'failed',
+                'error': str(e),
+                'semantic_insights': [],
+                'code_patterns': [],
+                'knowledge_graph': {},
+                'document_count': 0,
+                'embedding_dimension': 0
+            }
+        
+        # Step 3: Generate documentation
         try:
-            logger.info("Step 2: Generating documentation")
-            documentation = doc_generator.generate_documentation(analysis_result)
+            logger.info("Step 3: Generating documentation")
+            documentation = doc_generator.generate_documentation(analysis_result, rag_result)
             logger.info("Documentation generation completed successfully")
         except Exception as e:
             logger.error(f"Documentation generation failed: {e}")

diagnostic_start.sh

@@ -43,10 +43,70 @@ wait_for_service() {
     return 1
 }
 
+# Function to setup Python environment
+setup_python_environment() {
+    echo -e "${BLUE}� Setting up Python environment...${NC}"
+    
+    # Check if virtual environment exists
+    if [ ! -d ".venv" ]; then
+        echo -e "${YELLOW}📦 Creating virtual environment...${NC}"
+        python3 -m venv .venv
+        if [ $? -ne 0 ]; then
+            echo -e "${RED}❌ Failed to create virtual environment${NC}"
+            return 1
+        fi
+    fi
+    
+    # Activate virtual environment
+    source .venv/bin/activate
+    
+    # Upgrade pip
+    echo -e "${YELLOW}📦 Upgrading pip...${NC}"
+    pip install --upgrade pip
+    
+    # Install essential dependencies first
+    echo -e "${YELLOW}📦 Installing essential dependencies...${NC}"
+    pip install flask flask-cors flask-sqlalchemy python-dotenv requests
+    
+    if [ $? -eq 0 ]; then
+        echo -e "${GREEN}✅ Essential dependencies installed${NC}"
+        
+        # Test Flask import
+        python -c "import flask; print('✅ Flask version:', flask.__version__)" 2>/dev/null
+        if [ $? -eq 0 ]; then
+            echo -e "${GREEN}✅ Flask import test successful${NC}"
+        else
+            echo -e "${RED}❌ Flask import test failed${NC}"
+            return 1
+        fi
+        
+        # Install remaining dependencies if requirements.txt exists
+        if [ -f "requirements.txt" ]; then
+            echo -e "${YELLOW}📦 Installing remaining dependencies from requirements.txt...${NC}"
+            pip install -r requirements.txt
+        fi
+        
+        return 0
+    else
+        echo -e "${RED}❌ Failed to install essential dependencies${NC}"
+        return 1
+    fi
+}
+
 # Check environment configuration
 echo -e "${BLUE}🔍 Environment Configuration:${NC}"
 cd /Users/mantejsingh/Desktop/Documentation.AI
-python -c "
+
+# Setup Python environment
+if setup_python_environment; then
+    PYTHON_CMD=".venv/bin/python"
+    echo -e "${GREEN}✅ Using virtual environment Python${NC}"
+else
+    PYTHON_CMD="python3"
+    echo -e "${YELLOW}⚠️  Falling back to system Python${NC}"
+fi
+
+$PYTHON_CMD -c "
 import os
 from dotenv import load_dotenv
 load_dotenv()
@@ -89,9 +149,18 @@ fi
 
 echo ""
 echo -e "${BLUE}🔍 Dependencies Check:${NC}"
-if command -v python &> /dev/null; then
-    python_version=$(python --version 2>&1)
+if command -v $PYTHON_CMD &> /dev/null; then
+    python_version=$($PYTHON_CMD --version 2>&1)
     echo -e "${GREEN}✅ Python: $python_version${NC}"
+    
+    # Test Flask import
+    if $PYTHON_CMD -c "import flask" 2>/dev/null; then
+        flask_version=$($PYTHON_CMD -c "import flask; print(flask.__version__)" 2>/dev/null)
+        echo -e "${GREEN}✅ Flask: $flask_version${NC}"
+    else
+        echo -e "${RED}❌ Flask not installed${NC}"
+        echo -e "${YELLOW}💡 Run: pip install flask or use virtual environment${NC}"
+    fi
 else
     echo -e "${RED}❌ Python not found${NC}"
     exit 1
@@ -136,7 +205,7 @@ if [[ $REPLY =~ ^[Yy]$ ]]; then
 
     # Start backend
     echo -e "${BLUE}🎯 Starting backend server...${NC}"
-    python app.py > backend.log 2>&1 &
+    $PYTHON_CMD app.py > backend.log 2>&1 &
     BACKEND_PID=$!
 
     # Wait for backend to be ready
@@ -145,7 +214,7 @@ if [[ $REPLY =~ ^[Yy]$ ]]; then
 
         # Test backend health
         echo -e "${BLUE}🔬 Testing backend health...${NC}"
-        curl -s http://localhost:5002/api/health | python -m json.tool 2>/dev/null || echo "Health check returned non-JSON response"
+        curl -s http://localhost:5002/api/health | $PYTHON_CMD -m json.tool 2>/dev/null || echo "Health check returned non-JSON response"
 
     else
         echo -e "${RED}❌ Backend failed to start. Check backend.log:${NC}"
@@ -204,6 +273,6 @@ else
     echo -e "${YELLOW}✋ Startup cancelled. You can run this script again when ready.${NC}"
     echo ""
     echo -e "${BLUE}🔧 Manual startup commands:${NC}"
-    echo -e "   Backend: python app.py"
+    echo -e "   Backend: $PYTHON_CMD app.py"
     echo -e "   Frontend: cd frontend && npm start"
 fi

quick_setup.sh

@@ -0,0 +1,10 @@
+#!/bin/bash
+echo "Creating virtual environment..."
+cd /Users/mantejsingh/Desktop/Documentation.AI
+python3 -m venv .venv
+echo "Activating virtual environment..."
+source .venv/bin/activate
+echo "Installing Flask..."
+pip install flask
+echo "Testing Flask..."
+python -c "import flask; print('Flask installed successfully')"

sample-enhanced-docs.md

@@ -0,0 +1,95 @@
+# Sample Documentation Enhancement
+
+## 🚀 Enhanced GitHub-Style Documentation
+
+This is an example of enhanced markdown rendering with proper GitHub-style formatting.
+
+### Features
+
+- **Syntax Highlighting**: Code blocks are properly highlighted
+- **Interactive Elements**: Tables, links, and images are rendered beautifully
+- **Responsive Design**: Works perfectly on all screen sizes
+
+### Code Example
+
+```javascript
+// Example React component
+import React from 'react';
+
+const MyComponent = ({ title }) => {
+  return (
+    <div className="container">
+      <h1>{title}</h1>
+      <p>This is a sample component</p>
+    </div>
+  );
+};
+
+export default MyComponent;
+```
+
+### Python Example
+
+```python
+def analyze_repository(repo_url):
+    """
+    Analyze a GitHub repository and generate documentation.
+    
+    Args:
+        repo_url (str): The GitHub repository URL
+        
+    Returns:
+        dict: Analysis results
+    """
+    print(f"Analyzing repository: {repo_url}")
+    return {"status": "success", "data": []}
+```
+
+### Installation
+
+```bash
+# Install dependencies
+npm install
+
+# Start development server
+npm start
+
+# Build for production
+npm run build
+```
+
+### Configuration Table
+
+| Option | Type | Default | Description |
+|--------|------|---------|-------------|
+| `theme` | string | `"light"` | Application theme |
+| `autoSave` | boolean | `true` | Enable auto-save |
+| `maxFiles` | number | `100` | Maximum files to process |
+
+### Links and References
+
+- [GitHub Repository](https://github.com/example/repo)
+- [Documentation](https://docs.example.com)
+- [API Reference](https://api.example.com)
+
+### Blockquote Example
+
+> This is an important note about the implementation. 
+> 
+> Make sure to follow the security guidelines when implementing authentication.
+
+### Task List
+
+- [x] Setup project structure
+- [x] Implement basic features
+- [ ] Add advanced features
+- [ ] Write comprehensive tests
+- [ ] Deploy to production
+
+### Images
+
+![Architecture Diagram](https://via.placeholder.com/600x300/4F46E5/FFFFFF?text=Architecture+Diagram)
+
+---
+
+*This documentation is auto-generated by Documentation.AI*