Update app.py

app.py

@@ -1,117 +1,69 @@
 import os
-import sys
 import streamlit as st
-from huggingface_hub import InferenceClient
 
-# -------------------------------------------------
-# 1. Model Selection (DeepSeek-R1-Distill-Qwen-7B)
-# -------------------------------------------------
-MODEL_NAME = "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B"
-client = InferenceClient(MODEL_NAME)
-
-# -------------------------------------------------
-# 2. Load Roadmap and Rules
-# -------------------------------------------------
-def load_file(filename):
-    """Reads a text file and returns its content."""
+# Function to load files (roadmap and rules)
+def load_file(file_name):
+    """Reads the content of a given text file."""
     try:
-        with open(filename, "r", encoding="utf-8") as file:
+        with open(file_name, 'r', encoding='utf-8') as file:
             return file.read()
     except FileNotFoundError:
-        return f"⚠️ Error: {filename} not found!"
-
-# Load project guidance files
-roadmap_content = load_file("roadmap.txt")
-rules_content = load_file("rules.txt")
-
-# -------------------------------------------------
-# 3. Chatbot Response Function (With Project Guidance)
-# -------------------------------------------------
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    temperature,
-    top_p,
-):
-    """
-    Generates an AI response while considering project roadmap and rules.
-    """
-    # System message includes roadmap and rules for AI context
-    full_system_message = f"{system_message}\n\nProject Roadmap:\n{roadmap_content}\n\nProject Rules:\n{rules_content}"
-
-    messages = [{"role": "system", "content": full_system_message}]
-
-    for val in history:
-        if val[0]:  # User message
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:  # AI response
-            messages.append({"role": "assistant", "content": val[1]})
-
-    messages.append({"role": "user", "content": message})
-
-    response = ""
+        return f"Error: {file_name} not found!"
 
+# Function to update memory files (optional feature)
+def update_memory(file_name, new_content):
+    """Append new content to the given memory file (roadmap or rules)."""
     try:
-        for message in client.chat_completion(
-            messages,
-            max_tokens=None,  # 🚀 No token limit
-            stream=True,
-            temperature=temperature,
-            top_p=top_p,
-        ):
-            token = message.choices[0].delta.content
-
-            if token:
-                response += token
-                yield response  # Stream response live
+        with open(file_name, 'a', encoding='utf-8') as file:
+            file.write(new_content + "\n")
+        return f"Successfully updated {file_name}."
     except Exception as e:
-        yield f"⚠️ Error: {str(e)}"  # Show error messages
+        return f"Error updating {file_name}: {e}"
 
-# -------------------------------------------------
-# 4. Set up the Streamlit UI
-# -------------------------------------------------
+# Load files into memory (roadmap and rules)
+roadmap_content = load_file("roadmap.txt")
+rules_content = load_file("rules.txt")
 
+# Function to generate AI response based on user input
+def generate_response(user_input):
+    # Check if the user asks about the roadmap
+    if "roadmap" in user_input.lower():
+        return roadmap_content
+    # Check if the user asks about the rules
+    elif "rules" in user_input.lower():
+        return rules_content
+    # Handle next step or task-based queries
+    elif "next step" in user_input.lower():
+        return "The next step is: [fetch from roadmap]"
+    # Handle update-related queries
+    elif "update" in user_input.lower():
+        return "Updating memory..."  # Placeholder for functionality to update files
+    else:
+        return "I can help with the roadmap or rules. Just ask me!"
+
+# Streamlit UI setup
 st.title("Project Guidance AI Chatbot")
-st.write("Chat with the AI to get project guidance based on the roadmap and rules.")
-
-# Display the roadmap and rules text
-st.subheader("Project Roadmap:")
-st.write(roadmap_content)
-st.subheader("Project Rules:")
-st.write(rules_content)
 
-# -----------------------------------
-# Create a chat history state
-# -----------------------------------
-if "history" not in st.session_state:
-    st.session_state.history = []
+# Displaying a brief description of how the chatbot works
+st.write("""
+This chatbot helps you navigate through the project using the **roadmap.txt** and **rules.txt** as its memory.
+You can ask about the project steps, rules, or any guidance you need.
+""")
 
-# -----------------------------------
-# Input and Chat Display
-# -----------------------------------
-user_input = st.text_input("You:", "")
+# User input field for the chatbot
+user_input = st.text_input("Ask me anything about the project:")
 
+# Process user input and display the response from the AI
 if user_input:
-    st.session_state.history.append(("You", user_input))
-
-    # Set system message (helps guide the conversation)
-    system_message = "You are an AI assistant helping with project guidance based on a roadmap and rules."
-
-    # Generate response
-    ai_response = ""
-    for resp in respond(user_input, st.session_state.history, system_message, temperature=0.7, top_p=0.95):
-        ai_response = resp
-
-    # Update chat history with AI response
-    st.session_state.history.append(("AI", ai_response))
-
-# -----------------------------------
-# Display the chat history
-# -----------------------------------
-if st.session_state.history:
-    for chat in st.session_state.history:
-        if chat[0] == "You":
-            st.write(f"**You**: {chat[1]}")
-        else:
-            st.write(f"**AI**: {chat[1]}")
+    response = generate_response(user_input)
+    st.write(f"**AI Response:** {response}")
+
+    # Optional: update the memory files if user wants to add something
+    if "add to roadmap" in user_input.lower():
+        new_content = user_input.replace("add to roadmap", "").strip()
+        update_response = update_memory("roadmap.txt", new_content)
+        st.write(update_response)
+    if "add to rules" in user_input.lower():
+        new_content = user_input.replace("add to rules", "").strip()
+        update_response = update_memory("rules.txt", new_content)
+        st.write(update_response)