Initial commit

Dockerfile

@@ -0,0 +1,18 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+# Copy requirements first for better caching
+COPY requirements.txt .
+
+# Install dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application code
+COPY main.py .
+
+# Expose port 7860 (Hugging Face Spaces default)
+EXPOSE 7860
+
+# Run the application
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,40 @@
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+import wikipedia
+
+app = FastAPI()
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.get("/")
+async def get_wikipedia_url(topic: str):
+    try:
+        # Search for the page
+        page = wikipedia.page(topic, auto_suggest=True)
+        # Get the URL from the page object
+        url = page.url
+        return {"topic": topic, "url": url}
+    except wikipedia.exceptions.DisambiguationError as e:
+        # If multiple pages exist, take the first option
+        try:
+            page = wikipedia.page(e.options[0])
+            return {"topic": topic, "url": page.url}
+        except:
+            return {"topic": topic, "url": "https://en.wikipedia.org/wiki/Main_Page"}
+    except wikipedia.exceptions.PageError:
+        # Page doesn't exist
+        return {"topic": topic, "url": "https://en.wikipedia.org/wiki/Main_Page"}
+    except Exception as e:
+        # Any other error
+        return {"topic": topic, "url": "https://en.wikipedia.org/wiki/Main_Page"}
+
+@app.get("/health")
+async def health_check():
+    return {"status": "healthy"}

requirements.txt

@@ -0,0 +1,3 @@
+fastapi==0.104.1
+uvicorn==0.24.0
+wikipedia==1.4.0