ajnx014 commited on
Commit
f7a34c2
Β·
verified Β·
1 Parent(s): 35f39f9

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +50 -0
app.py ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import os
3
+ import requests
4
+
5
+ from langchain_community.document_loaders import WebBaseLoader
6
+ from langchain_text_splitters import RecursiveCharacterTextSplitter
7
+ from langchain_community.embeddings import HuggingFaceEmbeddings
8
+ from langchain_community.vectorstores import FAISS
9
+ from langchain.chains import RetrievalQA
10
+ from langchain.chat_models import ChatOpenAI
11
+
12
+ # βœ… Read OpenRouter API key from HF secret
13
+ OPENROUTER_API_KEY = os.environ.get("ArjunHF")
14
+
15
+ class OpenRouterChatModel(ChatOpenAI):
16
+ def __init__(self, **kwargs):
17
+ super().__init__(
18
+ openai_api_base="https://openrouter.ai/api/v1",
19
+ openai_api_key=OPENROUTER_API_KEY,
20
+ model_name="mistralai/mistral-small-3.2-24b-instruct:free",
21
+ **kwargs
22
+ )
23
+
24
+ def qa_on_url(url, question):
25
+ try:
26
+ loader = WebBaseLoader(url)
27
+ docs = loader.load()
28
+ splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
29
+ split_docs = splitter.split_documents(docs)
30
+
31
+ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
32
+ vectordb = FAISS.from_documents(split_docs, embeddings)
33
+ retriever = vectordb.as_retriever()
34
+
35
+ llm = OpenRouterChatModel(temperature=0.2)
36
+ qa_chain = RetrievalQA.from_chain_type(llm, retriever=retriever)
37
+ return qa_chain.run(question)
38
+ except Exception as e:
39
+ return f"❌ Error: {e}"
40
+
41
+ iface = gr.Interface(
42
+ fn=qa_on_url,
43
+ inputs=[gr.Textbox(label="Enter Web URL"), gr.Textbox(label="Your Question")],
44
+ outputs="text",
45
+ title="πŸ”Ž Ask Questions About Any Webpage (Mistral 3.2 via OpenRouter + LangChain)",
46
+ description="⚠️ This may take 10–20 seconds depending on the page length and LLM response time. Please be patient!"
47
+ )
48
+
49
+ if __name__ == "__main__":
50
+ iface.launch()