Create app.py

app.py

@@ -0,0 +1,62 @@
+import streamlit as st
+from transformers import pipeline
+from googlesearch import search
+
+# Function to load the model
+@st.cache_resource
+def load_model():
+    model_name = "nisten/Biggie-SmoLlm-0.15B-Base"  # Replace with your preferred model
+    generator = pipeline("text-generation", model=model_name)
+    return generator
+
+# Load the model once
+generator = load_model()
+
+# Function to process the query using the open-source LLM for general chat
+def chat_with_llm(query):
+    prompt = f"User: {query}\nAssistant:"
+    response = generator(prompt, max_length=100, num_return_sequences=1)
+    return response[0]['generated_text'].strip()
+
+# Function to process the query for search intent
+def process_query_with_llm(query):
+    prompt = f"User asked: '{query}'. What would be the best search query to use?"
+    response = generator(prompt, max_length=50, num_return_sequences=1)
+    return response[0]['generated_text'].strip()
+
+# Function to perform a Google search using the googlesearch-python package
+def search_web(query):
+    search_results = []
+    for result in search(query, num_results=3):
+        search_results.append(result)
+    return search_results
+
+# Streamlit UI
+st.title("Interactive Chatbot")
+
+# Input field for user query
+user_input = st.text_input("You:", "")
+
+if user_input:
+    st.write(f"**You:** {user_input}")
+    
+    # Determine if the query is a search or a general chat
+    if any(keyword in user_input.lower() for keyword in ["search", "find", "get me", "give me"]):
+        # If the user input indicates a search intent
+        search_query = process_query_with_llm(user_input)
+        st.write(f"**Processed Query:** {search_query}")
+
+        # Search the web using the processed query
+        links = search_web(search_query)
+
+        # Display the search results
+        if links:
+            st.write("Here are some links you might find useful:")
+            for idx, link in enumerate(links):
+                st.write(f"{idx + 1}. {link}")
+        else:
+            st.write("Sorry, I couldn't find any relevant links.")
+    else:
+        # Handle general conversation
+        response = chat_with_llm(user_input)
+        st.write(f"**Chatbot:** {response}")