Deploy TechTales AI on Hugging Face

README.md

@@ -1,13 +1,61 @@
+# 🤖 TechTales AI - Your AI-Powered Assistant  
+
+
+TechTales AI is an AI-powered search assistant built using LangChain and Claude API. This project allows users to ask any questions and receive structured responses. Follow the steps below to set up and run the application.
+
+
+## 🚀 Features
+- Uses **Claude-3-5-Sonnet** for intelligent responses.
+- Supports **tool integration** (search, wiki, save tools).
+- Built with **Streamlit** for an interactive UI.
+- Secure API key handling through **.env file**.
+
 ---
-title: TechTales AI
-emoji: 🚀
-colorFrom: gray
-colorTo: purple
-sdk: streamlit
-sdk_version: 1.44.0
-app_file: app.py
-pinned: false
-short_description: An AI-powered assistant built using LangChain and Claud API
+
+## 🛠️ Installation & Setup
+
+### **1️⃣ Clone the Repository**
+```bash
+git clone https://huggingface.co/spaces/YourHuggingFaceRepo/TechTales-AI.git
+cd TechTales-AI
+```
+
+### **2️⃣ Create a Virtual Environment (Optional but Recommended)**
+```bash
+python -m venv venv
+source venv/bin/activate  # On Mac/Linux
+venv\Scripts\activate    # On Windows
+```
+
+### **3️⃣ Install Dependencies**
+```bash
+pip install -r requirements.txt
+```
+
+### **4️⃣ Set Up API Keys**
+This project uses the **Claude API**. You need to create a `.env` file to store your API key.
+
+#### **Create a `.env` file in the project root and add:**
+```env
+ANTHROPIC_API_KEY=your_claude_api_key_here
+```
+💡 **Note:** You must obtain your own API key from [Anthropic](https://www.anthropic.com/) as the key is not provided in this repository.
+
+### **5️⃣ Run the Application**
+```bash
+streamlit run app.py
+```
+
+This will start a local server, and you can interact with TechTales-AI via the browser.
+
+## 🔥 Contributing
+Feel free to fork the repo and submit pull requests to improve TechTales AI! 😊
+
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+## 📄 License
+This project is licensed under the MIT License.
+
+Happy coding! 🚀
+
+

TestModelNames.py

@@ -0,0 +1,5 @@
+import anthropic
+
+client = anthropic.Anthropic(api_key="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")
+models = client.models.list()
+print(models)  # This will show the exact API-compatible model names

main.py

@@ -0,0 +1,91 @@
+import streamlit as st
+from dotenv import load_dotenv
+from pydantic import BaseModel
+from langchain_anthropic import ChatAnthropic
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import PydanticOutputParser
+from langchain.agents import create_tool_calling_agent, AgentExecutor
+from tools import search_tool, wiki_tool, save_tool
+
+# Load environment variables
+load_dotenv()
+
+# Define the response format
+class ResearchResponse(BaseModel):
+    topic: str
+    summary: str
+    sources: list[str]
+    tools_used: list[str]
+
+# Initialize Claude model with low token usage
+llm = ChatAnthropic(model="claude-3-5-sonnet-20241022", max_tokens=300)
+
+# Create parser
+parser = PydanticOutputParser(pydantic_object=ResearchResponse)
+
+# Define prompt template
+prompt = ChatPromptTemplate.from_messages(
+    [
+        (
+            "system",
+            """
+            You are an AI assistant that will help of any general questions.
+            Answer the user query and use necessary tools. Don't answer any offensive questions or any swearings.
+            Wrap the output in this format and provide no other text\n{format_instructions}
+            """,
+        ),
+        ("placeholder", "{chat_history}"),
+        ("human", "{query}"),
+        ("placeholder", "{agent_scratchpad}"),
+    ]
+).partial(format_instructions=parser.get_format_instructions())
+
+# Define tools and create the agent
+tools = [search_tool, wiki_tool, save_tool]
+agent = create_tool_calling_agent(llm=llm, prompt=prompt, tools=tools)
+
+# Initialize the agent executor
+agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=False)
+
+# Streamlit UI
+st.set_page_config(page_title="TechTales AI - Your own AI Powered Assistant", layout="centered")
+st.title("💡 TechTales AI - Your own AI Powered Assistant")
+
+# Initialize chat history in session state
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Display chat history
+for msg in st.session_state.messages:
+    st.chat_message(msg["role"]).write(msg["content"])
+
+# Chat input with Enter button
+query = st.text_input("I am your personal Assistant, Ask me anything...", key="query_input")
+submit_button = st.button("Enter")
+
+if submit_button and query:
+    # Add user message to history
+    st.session_state.messages.append({"role": "user", "content": query})
+    st.chat_message("user").write(query)
+
+    # Get AI response
+    raw_response = agent_executor.invoke({"query": query})
+
+    try:
+        structured_response = parser.parse(raw_response.get("output")[0]["text"])
+        response_text = f"**Topic:** {structured_response.topic}\n\n**Summary:** {structured_response.summary}\n\n**Sources:** {', '.join(structured_response.sources)}"
+    except Exception:
+        response_text = "I'm sorry, I couldn't process that request."
+
+    # Add AI response to history
+    st.session_state.messages.append({"role": "assistant", "content": response_text})
+    st.chat_message("assistant").write(response_text)
+
+# Clear Chat Button
+if st.button("🗑️ Clear Chat"):
+    st.session_state.messages = []
+    st.rerun()  # Corrected method
+
+# Developer Name Display
+st.markdown("---")
+st.markdown("**Developed by: Pankaj Kumar**", unsafe_allow_html=True)

requirements.txt

@@ -0,0 +1,10 @@
+langchain 
+wikipedia 
+langchain-community 
+langchain-openai
+langchain-anthropic
+python-dotenv
+pydantic
+duckduckgo-search
+streamlit
+streamlit-chat

research_output.txt

@@ -0,0 +1,22 @@
+--- Research Output ---
+Timestamp: 2025-03-24 17:17:52
+
+Capital of France Research:
+
+Capital City: Paris
+Country: France
+Official Status: Capital since 1944
+
+Key Statistics:
+- City Area: 105 km² (41 sq mi)
+- City Population: 2,048,472 (2025 estimate)
+- Metropolitan Population: 12,271,794 (2023)
+- Percentage of National Population: ~19%
+
+Historical Significance:
+- Major global center since 17th century
+- Known as "City of Light" since 19th century
+- Centers of: finance, diplomacy, commerce, culture, fashion, gastronomy
+
+Location: Center of Île-de-France region
+

tools.py

@@ -0,0 +1,30 @@
+from langchain_community.tools import WikipediaQueryRun, DuckDuckGoSearchRun
+from langchain_community.utilities import WikipediaAPIWrapper
+from langchain.tools import Tool
+from datetime import datetime
+
+def save_to_txt(data: str, filename: str = "research_output.txt"):
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    formatted_text = f"--- Research Output ---\nTimestamp: {timestamp}\n\n{data}\n\n"
+
+    with open(filename, "a", encoding="utf-8") as f:
+        f.write(formatted_text)
+    
+    return f"Data successfully saved to {filename}"
+
+save_tool = Tool(
+    name="save_text_to_file",
+    func=save_to_txt,
+    description="Saves structured research data to a text file.",
+)
+
+search = DuckDuckGoSearchRun()
+search_tool = Tool(
+    name="search",
+    func=search.run,
+    description="Search the web for information",
+)
+
+api_wrapper = WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=1000)
+wiki_tool = WikipediaQueryRun(api_wrapper=api_wrapper)
+