C

.gitignore

@@ -0,0 +1,8 @@
+.env
+.DS_Store
+
+model_cache
+cache
+__pycache__
+
+

alembic.ini

@@ -1,64 +0,0 @@
-# A generic, single database configuration.
-
-[alembic]
-# Path to migration scripts
-script_location = alembic
-
-# Template used to generate migration files
-# file_template = %%(rev)s_%%(slug)s
-
-# Sys.path path, will be prepended to sys.path if present.
-# Defaults to the current working directory.
-prepend_sys_path = .
-
-# Version path separator
-version_path_separator = os  # Use os.pathsep. Default configuration used for new projects.
-
-# Database connection string (SQLite persistent database)
-sqlalchemy.url = sqlite+aiosqlite:///./sql_app.db
-
-# Explicitly set pooling to avoid unnecessary resets
-[sqlalchemy]
-poolclass = QueuePool
-
-# Post-migration hooks (optional, can be customized)
-[post_write_hooks]
-# hooks = black
-# black.type = console_scripts
-# black.entrypoint = black
-# black.options = -l 79 REVISION_SCRIPT_FILENAME
-
-# Logging configuration
-[loggers]
-keys = root,sqlalchemy,alembic
-
-[handlers]
-keys = console
-
-[formatters]
-keys = generic
-
-[logger_root]
-level = WARN
-handlers = console
-qualname =
-
-[logger_sqlalchemy]
-level = WARN
-handlers =
-qualname = sqlalchemy.engine
-
-[logger_alembic]
-level = INFO
-handlers =
-qualname = alembic
-
-[handler_console]
-class = StreamHandler
-args = (sys.stderr,)
-level = NOTSET
-formatter = generic
-
-[formatter_generic]
-format = %(levelname)-5.5s [%(name)s] %(message)s
-datefmt = %H:%M:%S

app.log

@@ -1,22 +0,0 @@
-2025-02-28 18:27:16,754 - INFO - Chat request received with 1 messages
-2025-02-28 18:27:16,755 - INFO - Formatted prompt: give me saas app
-
-2025-02-28 18:27:29,908 - INFO - Generated response: i want to download saas app
-2025-02-28 18:28:08,376 - INFO - Feature generation request received with requirements: gerneate me 10 feature to use in saas
-2025-02-28 18:28:08,377 - INFO - Generated prompt: Generate 5 features based on the following requirements. Format each feature as a JSON object with 'feature' and 'short_description' fields.
-
-Requirements: gerneate me 10 feature to use in saas
-2025-02-28 18:29:29,392 - INFO - Model response: i would like 10 feature to use in saas
-2025-02-28 18:29:29,392 - INFO - Returning features: [Feature(feature='Feature 1', short_description='Description 1'), Feature(feature='Feature 2', short_description='Description 2')]
-2025-02-28 18:31:00,786 - INFO - Feature generation request received with requirements: gerneate me 10 feature to use in saas
-2025-02-28 18:31:00,787 - INFO - Generated prompt: Generate 5 detailed SaaS features based on the following requirements. Each feature should be practical and implementation-ready. Format your response as a list of JSON objects, each with 'feature' and 'short_description' fields.
-
-Example format:
-{
-  'feature': 'User Authentication',
-  'short_description': 'Secure login system with OAuth2 and MFA support'
-}
-
-Requirements: gerneate me 10 feature to use in saas
-
-Provide 5 features in the exact JSON format shown above.

app/controllers/f5_model.py

@@ -8,43 +8,48 @@ class F5ModelHandler:
     def __init__(self):
         logging.info("Initializing F5ModelHandler...")
         try:
-            logging.info("Loading model 'google/flan-t5-small'...")
-            self.model_name = "google/flan-t5-small"
-            # Use pipeline for simpler model loading
+            # Using a larger T5 model for better generation
+            logging.info("Loading model 'google/flan-t5-large'...")
+            self.model_name = "google/flan-t5-large"  # Upgraded from small to large
+            
+            # Use pipeline for simpler model loading with better parameters
             self.generator = pipeline(
                 "text2text-generation",
                 model=self.model_name,
                 device="cuda" if torch.cuda.is_available() else "cpu",
-                torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+                torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+                model_kwargs={"cache_dir": "./model_cache"}  # Cache the model locally
             )
             logging.info(f"Model loaded successfully on {self.generator.device}")
         except Exception as e:
             logging.error(f"Error loading model: {str(e)}")
             raise
         
-    async def generate_response(self, prompt: str, max_length: int = 2048) -> str:
+    async def generate_response(self, prompt: str, max_length: int = 3072) -> str:
         try:
             logging.info(f"Generating response for prompt: {prompt[:100]}...")
             
-            # Generate with more focused parameters
+            # Enhanced generation parameters for better quality
             response = self.generator(
                 prompt,
                 max_length=max_length,
+                min_length=1000,  # Ensure minimum length
                 num_beams=5,
-                temperature=0.7,
+                temperature=0.8,
                 top_p=0.95,
                 top_k=50,
                 repetition_penalty=1.2,
                 length_penalty=1.0,
                 do_sample=True,
-                num_return_sequences=1
+                num_return_sequences=1,
+                early_stopping=True
             )[0]['generated_text']
             
             # Clean up the response
             response = response.strip()
             
             # Ensure minimum content length
-            if len(response) < 100:
+            if len(response) < 500:  # Increased minimum length requirement
                 logging.warning("Response too short, regenerating...")
                 return await self.generate_response(prompt, max_length)
             
@@ -55,14 +60,15 @@ class F5ModelHandler:
             logging.error(f"Error generating response: {str(e)}")
             raise
 
-    async def stream_response(self, prompt: str, max_length: int = 1000):
+    async def stream_response(self, prompt: str, max_length: int = 3072):
         try:
             response = self.generator(
                 prompt,
                 max_length=max_length,
+                min_length=1000,
                 num_beams=4,
-                temperature=0.7,
-                top_p=0.9,
+                temperature=0.8,
+                top_p=0.95,
                 do_sample=True,
                 return_full_text=False
             )[0]['generated_text']

app/main.py

@@ -190,89 +190,43 @@ async def generate_plan(request: ProjectPlanRequest):
     try:
         logger.info(f"Plan generation request received for project: {request.project_title}")
         
-        # Enhanced prompt for more detailed output
-        prompt = f"""Generate a comprehensive technical project plan for '{request.project_title}' with detailed sections.
-        Include specific {request.platform} services and implementation details in each section.
+        # Simplified prompt for better generation
+        prompt = f"""Create a detailed technical project plan for a {request.platform} project named '{request.project_title}'.
 
-        1. Executive Summary
-        - Provide a detailed overview of the project (4-5 paragraphs)
-        - Include project goals, target users, and key outcomes
-        - Highlight main {request.platform} services to be used
-        - Explain expected business impact
+Requirements: {request.requirements}
+Features: {', '.join(request.features)}
+Additional Requirements: {request.additional_requirements}
 
-        2. Project Scope and Objectives
-        - Define detailed scope including:
-          * Core functionalities
-          * System boundaries
-          * Integration points
-        - List at least 5 specific objectives
-        - Include measurable success criteria
+Generate a complete plan with the following sections:
 
-        3. Architecture Overview
-        - Detailed {request.platform} architecture including:
-          * Frontend architecture
-          * Backend services
-          * Database design
-          * Integration patterns
-          * Network topology
-        - Explain how components interact
-        - Include scalability considerations
+1. Executive Summary
+Brief overview of the project, its goals, and business impact.
 
-        4. Component Design
-        - Detail at least 3 core components with:
-          * Purpose and functionality
-          * Technical specifications
-          * Data flow
-          * Integration points
-          * Performance requirements
+2. Project Scope and Objectives
+Core functionalities and specific objectives.
 
-        5. Security and Compliance
-        - List specific {request.platform} security services
-        - Detail authentication and authorization
-        - Describe data protection measures
-        - Include compliance requirements
-        - Specify security monitoring
+3. Architecture Overview
+{request.platform} architecture components and their interactions.
 
-        6. Deployment, Testing, and Monitoring
-        - Detailed CI/CD pipeline
-        - Test strategy including:
-          * Unit testing
-          * Integration testing
-          * Performance testing
-        - Monitoring setup with {request.platform} tools
-        - Alerting and logging strategy
+4. Component Design
+Core components and their technical specifications.
 
-        7. Team Roles and Skills
-        - List all required roles
-        - Detail specific skills needed
-        - Include {request.platform} certifications
-        - Define team structure
-        - Specify responsibilities
+5. Security and Compliance
+Security measures and compliance requirements.
 
-        8. Cost Estimates and Optimization
-        - Break down {request.platform} service costs
-        - Resource optimization strategies
-        - Scaling considerations
-        - Cost monitoring approach
-        - Budget optimization tips
+6. Deployment, Testing, and Monitoring
+Implementation approach and quality assurance.
 
-        9. Project Tasks and Milestones
-        - Detailed project phases
-        - Specific tasks for each phase
-        - Timeline estimates
-        - Dependencies
-        - Critical path items
+7. Team Roles and Skills
+Required team members and their expertise.
 
-        Project Requirements:
-        {request.requirements}
+8. Cost Estimates and Optimization
+Resource planning and cost optimization.
 
-        Features to implement:
-        {', '.join(request.features)}
+9. Project Tasks and Milestones
+Project phases and timeline.
 
-        Additional Requirements:
-        {request.additional_requirements}
-
-        Generate a detailed plan following this structure with specific {request.platform} implementation details."""
+Provide detailed content for each section."""
         
         logger.info("Generating plan with model...")
         response = await model_handler.generate_response(prompt)

app/templates/index.html

@@ -5,6 +5,7 @@
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>F5 Model Test Interface</title>
     <script src="https://cdn.tailwindcss.com"></script>
+    <link href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css" rel="stylesheet">
     <style>
         .loading {
             display: none;
@@ -40,6 +41,70 @@
             0%, 100% { opacity: 1; }
             50% { opacity: 0.5; }
         }
+
+        /* Chat Widget Styles */
+        .chat-widget {
+            position: fixed;
+            bottom: 20px;
+            right: 20px;
+            z-index: 1000;
+        }
+
+        .chat-button {
+            width: 60px;
+            height: 60px;
+            border-radius: 50%;
+            background-color: #3B82F6;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            cursor: pointer;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+            transition: all 0.3s ease;
+        }
+
+        .chat-button:hover {
+            transform: scale(1.1);
+        }
+
+        .chat-window {
+            position: fixed;
+            bottom: 90px;
+            right: 20px;
+            width: 350px;
+            height: 500px;
+            background: white;
+            border-radius: 12px;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+            display: none;
+            flex-direction: column;
+            overflow: hidden;
+        }
+
+        .chat-window.active {
+            display: flex;
+        }
+
+        .chat-header {
+            padding: 1rem;
+            background: #3B82F6;
+            color: white;
+            display: flex;
+            justify-content: space-between;
+            align-items: center;
+        }
+
+        .chat-messages {
+            flex-grow: 1;
+            overflow-y: auto;
+            padding: 1rem;
+        }
+
+        .chat-input-container {
+            padding: 1rem;
+            border-top: 1px solid #e5e7eb;
+            background: white;
+        }
     </style>
 </head>
 <body class="bg-gray-100">
@@ -87,30 +152,26 @@
                 <h3 class="text-lg font-semibold mb-2" id="planTitle"></h3>
                 <div id="planSections" class="space-y-4"></div>
             </div>
-        </div>
-
-        <!-- Chat Section -->
-        <div class="bg-white rounded-lg shadow-md p-6 mb-8">
-            <h2 class="text-xl font-semibold mb-4">Chat Test</h2>
-            <div class="mb-4">
-                <div id="chatHistory" class="bg-gray-50 rounded p-4 h-64 overflow-y-auto mb-4"></div>
-                <div id="chatLoading" class="loading mb-4">
-                    <div></div>
-                    <div></div>
-                    <div></div>
-                </div>
-                <div class="flex gap-2">
-                    <input type="text" id="chatInput" 
-                           class="flex-1 border rounded px-3 py-2" 
-                           placeholder="Type your message...">
-                    <button onclick="sendMessage()" 
-                            class="bg-blue-500 text-white px-4 py-2 rounded hover:bg-blue-600">
-                        Send
+            <div id="debugPanel" class="mt-6 bg-gray-800 rounded p-4 text-white hidden">
+                <div class="flex justify-between items-center mb-4">
+                    <h3 class="text-lg font-semibold">Debug Panel</h3>
+                    <button onclick="toggleDebugPanel()" class="text-gray-400 hover:text-white">
+                        <i class="fas fa-times"></i>
                     </button>
-                    <label class="flex items-center gap-2">
-                        <input type="checkbox" id="streamToggle">
-                        Stream
-                    </label>
+                </div>
+                <div class="space-y-4">
+                    <div>
+                        <h4 class="text-sm font-semibold text-gray-400">Request Data</h4>
+                        <pre id="requestData" class="text-xs bg-gray-900 p-2 rounded overflow-auto max-h-40"></pre>
+                    </div>
+                    <div>
+                        <h4 class="text-sm font-semibold text-gray-400">Raw API Response</h4>
+                        <pre id="rawResponse" class="text-xs bg-gray-900 p-2 rounded overflow-auto max-h-40"></pre>
+                    </div>
+                    <div>
+                        <h4 class="text-sm font-semibold text-gray-400">Parsed Sections</h4>
+                        <pre id="parsedSections" class="text-xs bg-gray-900 p-2 rounded overflow-auto max-h-40"></pre>
+                    </div>
                 </div>
             </div>
         </div>
@@ -136,22 +197,73 @@
         </div>
     </div>
 
+    <!-- Chat Widget -->
+    <div class="chat-widget">
+        <div class="chat-button" onclick="toggleChat()">
+            <i class="fas fa-comments text-white text-2xl"></i>
+        </div>
+        
+        <div class="chat-window" id="chatWindow">
+            <div class="chat-header">
+                <span class="font-semibold">Chat Assistant</span>
+                <button onclick="toggleChat()" class="text-white hover:text-gray-200">
+                    <i class="fas fa-times"></i>
+                </button>
+            </div>
+            
+            <div class="chat-messages" id="chatHistory">
+                <!-- Messages will be inserted here -->
+            </div>
+            
+            <div id="chatLoading" class="loading mx-auto my-2" style="transform: scale(0.5);">
+                <div></div>
+                <div></div>
+                <div></div>
+            </div>
+            
+            <div class="chat-input-container">
+                <div class="flex gap-2">
+                    <input type="text" id="chatInput" 
+                           class="flex-1 border rounded px-3 py-2" 
+                           placeholder="Type your message...">
+                    <button onclick="sendMessage()" 
+                            class="bg-blue-500 text-white px-4 py-2 rounded hover:bg-blue-600">
+                        <i class="fas fa-paper-plane"></i>
+                    </button>
+                </div>
+                <div class="flex items-center gap-2 mt-2">
+                    <label class="flex items-center gap-2 text-sm text-gray-600">
+                        <input type="checkbox" id="streamToggle">
+                        Stream responses
+                    </label>
+                </div>
+            </div>
+        </div>
+    </div>
+
     <script>
         let chatHistory = [];
+        let chatWindowOpen = false;
+
+        function toggleChat() {
+            const chatWindow = document.getElementById('chatWindow');
+            chatWindowOpen = !chatWindowOpen;
+            chatWindow.classList.toggle('active');
+        }
 
         async function sendMessage() {
             const input = document.getElementById('chatInput');
             const message = input.value.trim();
+            const streamMode = document.getElementById('streamToggle').checked;
+            
             if (!message) return;
-
-            // Show loading indicator
-            document.getElementById('chatLoading').style.display = 'block';
             
-            // Add user message to chat
-            appendMessage('user', message);
             input.value = '';
-
-            const streamMode = document.getElementById('streamToggle').checked;
+            appendMessage('user', message);
+            
+            const loading = document.getElementById('chatLoading');
+            loading.style.display = 'block';
+            
             try {
                 if (streamMode) {
                     const response = await fetch('/chat', {
@@ -195,10 +307,9 @@
                 }
             } catch (error) {
                 console.error('Error:', error);
-                appendMessage('system', 'Error: Failed to get response');
+                appendMessage('assistant', 'Sorry, there was an error processing your request.');
             } finally {
-                // Hide loading indicator
-                document.getElementById('chatLoading').style.display = 'none';
+                loading.style.display = 'none';
             }
         }
 
@@ -245,11 +356,12 @@
         function updateChatDisplay() {
             const chatDiv = document.getElementById('chatHistory');
             chatDiv.innerHTML = chatHistory.map(msg => `
-                <div class="mb-2">
-                    <strong class="${msg.role === 'user' ? 'text-blue-600' : 'text-green-600'}">
-                        ${msg.role}:
-                    </strong>
-                    <span class="ml-2">${msg.content}</span>
+                <div class="mb-3">
+                    <div class="flex items-start ${msg.role === 'user' ? 'justify-end' : 'justify-start'}">
+                        <div class="max-w-[80%] ${msg.role === 'user' ? 'bg-blue-500 text-white' : 'bg-gray-200'} rounded-lg px-4 py-2">
+                            <div class="text-sm">${msg.content}</div>
+                        </div>
+                    </div>
                 </div>
             `).join('');
             chatDiv.scrollTop = chatDiv.scrollHeight;
@@ -325,6 +437,11 @@
             }
         }
 
+        function toggleDebugPanel() {
+            const debugPanel = document.getElementById('debugPanel');
+            debugPanel.classList.toggle('hidden');
+        }
+
         function displayPlan(data) {
             console.log('Displaying plan data:', data);
             const output = document.getElementById('planOutput');
@@ -368,6 +485,15 @@
 
             console.log('Final HTML output:', sections.innerHTML);
             output.classList.remove('hidden');
+
+            // Update debug panel
+            document.getElementById('debugPanel').classList.remove('hidden');
+            document.getElementById('requestData').textContent = JSON.stringify({
+                project_title: data.project_title,
+                sections: data.sections
+            }, null, 2);
+            document.getElementById('rawResponse').textContent = data.raw_content;
+            document.getElementById('parsedSections').textContent = JSON.stringify(data.sections, null, 2);
         }
     </script>
 </body>

requirements.txt

@@ -1,4 +1,5 @@
 fastapi==0.109.0
+torch
 uvicorn==0.27.0
 sqlalchemy==2.0.25
 aiosqlite==0.19.0
@@ -12,4 +13,9 @@ passlib==1.7.4
 alembic==1.13.1
 openpyxl
 xlsxwriter
-pypdf
+pypdf
+transformers>=4.36.2
+torch
+accelerate>=0.25.0
+sentencepiece>=0.1.99
+