Add save data functionality and enhance search result model with model type

static/index.html

@@ -147,16 +147,19 @@ class LoginResponse {
                     <pre><code class="language-dart">class SearchResult {
   final String text;
   final double similarity;
+  final String modelType;
 
   SearchResult({
     required this.text,
     required this.similarity,
+    required this.modelType,
   });
 
   factory SearchResult.fromJson(Map<String, dynamic> json) {
     return SearchResult(
       text: json['text'],
       similarity: json['similarity'].toDouble(),
+      modelType: json['model_type'],
     );
   }
 }
@@ -191,18 +194,97 @@ Future<List<SearchResult>> search(String query, String accessToken) async {
                     <pre><code class="language-json">[
     {
         "text": "Result 1 text",
-        "similarity": 0.95
+        "similarity": 0.95,
+        "model_type": "all-mpnet-base-v2"
     },
     {
         "text": "Result 2 text",
-        "similarity": 0.92
-    },
-    {
-        "text": "Result 3 text",
-        "similarity": 0.89
+        "similarity": 0.92,
+        "model_type": "openai"
     }
 ]</code></pre>
                 </div>
+
+                <div class="endpoint" id="save">
+                    <h3>4. Save Data Endpoint</h3>
+                    <p><span class="method">POST</span><span class="endpoint-url">/save</span></p>
+                    <p>This endpoint is used to save user feedback and search results to the Hugging Face dataset. It requires a valid access token.</p>
+                    
+                    <h4>cURL Request:</h4>
+                    <pre><code class="language-bash">curl -X POST https://humblebeeai-al-ghazali-rag-retrieval-api.hf.space/save \
+     -H "Content-Type: application/json" \
+     -H "Authorization: Bearer your-access-token" \
+     -d '{
+           "items": [
+               {
+                   "user_type": "user",
+                   "username": "user123",
+                   "query": "test query",
+                   "retrieved_text": "Result 1 text",
+                   "model_type": "all-mpnet-base-v2",
+                   "reaction": "positive"
+               }
+           ]
+         }'</code></pre>
+
+                    <h4>Flutter Implementation:</h4>
+                    <pre><code class="language-dart">class SaveInput {
+  final String userType;
+  final String username;
+  final String query;
+  final String retrievedText;
+  final String modelType;
+  final String reaction;
+
+  SaveInput({
+    required this.userType,
+    required this.username,
+    required this.query,
+    required this.retrievedText,
+    required this.modelType,
+    required this.reaction,
+  });
+
+  Map<String, dynamic> toJson() {
+    return {
+      'user_type': userType,
+      'username': username,
+      'query': query,
+      'retrieved_text': retrievedText,
+      'model_type': modelType,
+      'reaction': reaction,
+    };
+  }
+}
+
+Future<void> saveData(List<SaveInput> items, String accessToken) async {
+  final url = Uri.parse('https://humblebeeai-al-ghazali-rag-retrieval-api.hf.space/save');
+  
+  try {
+    final response = await http.post(
+      url,
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': 'Bearer $accessToken',
+      },
+      body: jsonEncode({
+        'items': items.map((item) => item.toJson()).toList(),
+      }),
+    );
+
+    if (response.statusCode != 200) {
+      throw Exception('Failed to save data: ${response.body}');
+    }
+  } catch (e) {
+    throw Exception('Network error: $e');
+  }
+}</code></pre>
+                    
+                    <h4>Response:</h4>
+                    <pre><code class="language-json">{
+    "message": "Data saved successfully"
+}</code></pre>
+                </div>
             </section>
 
             <section id="workflow">
@@ -226,7 +308,26 @@ Future<List<SearchResult>> search(String query, String accessToken) async {
                 </div>
 
                 <div class="endpoint">
-                    <h3>Step 3: Refresh Token</h3>
+                    <h3>Step 3: Save Data</h3>
+                    <pre><code class="language-bash">curl -X POST https://humblebeeai-al-ghazali-rag-retrieval-api.hf.space/save \
+     -H "Content-Type: application/json" \
+     -H "Authorization: Bearer your-access-token" \
+     -d '{
+           "items": [
+               {
+                   "user_type": "user",
+                   "username": "user123",
+                   "query": "test query",
+                   "retrieved_text": "Result 1 text",
+                   "model_type": "all-mpnet-base-v2",
+                   "reaction": "positive"
+               }
+           ]
+         }'</code></pre>
+                </div>
+
+                <div class="endpoint">
+                    <h3>Step 4: Refresh Token</h3>
                     <pre><code class="language-bash">curl -X POST https://humblebeeai-al-ghazali-rag-retrieval-api.hf.space/refresh \
      -H "Content-Type: application/json" \
      -d '{"refresh_token": "your-refresh-token"}'</code></pre>
@@ -301,6 +402,30 @@ class _ApiWorkflowExampleState extends State<ApiWorkflowExample> {
     }
   }
 
+  Future<void> _saveData() async {
+    final items = [
+      SaveInput(
+        userType: 'user',
+        username: 'user123',
+        query: 'test query',
+        retrievedText: _searchResults[0].text,
+        modelType: _searchResults[0].modelType,
+        reaction: 'positive',
+      ),
+    ];
+
+    try {
+      await saveData(items, _accessToken);
+      ScaffoldMessenger.of(context).showSnackBar(
+        SnackBar(content: Text('Data saved successfully!')),
+      );
+    } catch (e) {
+      ScaffoldMessenger.of(context).showSnackBar(
+        SnackBar(content: Text('Failed to save data: $e')),
+      );
+    }
+  }
+
   Future<void> _refreshToken() async {
     try {
       final loginResponse = await refreshToken(_refreshToken);
@@ -336,6 +461,10 @@ class _ApiWorkflowExampleState extends State<ApiWorkflowExample> {
               onPressed: _search,
               child: Text('Search'),
             ),
+            ElevatedButton(
+              onPressed: _saveData,
+              child: Text('Save Data'),
+            ),
             ElevatedButton(
               onPressed: _refreshToken,
               child: Text('Refresh Token'),
@@ -347,7 +476,7 @@ class _ApiWorkflowExampleState extends State<ApiWorkflowExample> {
                   final result = _searchResults[index];
                   return ListTile(
                     title: Text(result.text),
-                    subtitle: Text('Similarity: ${result.similarity}'),
+                    subtitle: Text('Similarity: ${result.similarity}, Model: ${result.modelType}'),
                   );
                 },
               ),
@@ -430,6 +559,29 @@ Future<List<SearchResult>> search(String query, String accessToken) async {
   }
 }
 
+Future<void> saveData(List<SaveInput> items, String accessToken) async {
+  final url = Uri.parse('https://humblebeeai-al-ghazali-rag-retrieval-api.hf.space/save');
+
+  try {
+    final response = await http.post(
+      url,
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': 'Bearer $accessToken',
+      },
+      body: jsonEncode({
+        'items': items.map((item) => item.toJson()).toList(),
+      }),
+    );
+
+    if (response.statusCode != 200) {
+      throw Exception('Failed to save data: ${response.body}');
+    }
+  } catch (e) {
+    throw Exception('Network error: $e');
+  }
+}
+
 class LoginResponse {
   final String accessToken;
   final String refreshToken;
@@ -456,19 +608,53 @@ class LoginResponse {
 class SearchResult {
   final String text;
   final double similarity;
+  final String modelType;
 
   SearchResult({
     required this.text,
     required this.similarity,
+    required this.modelType,
   });
 
   factory SearchResult.fromJson(Map<String, dynamic> json) {
     return SearchResult(
       text: json['text'],
       similarity: json['similarity'].toDouble(),
+      modelType: json['model_type'],
     );
   }
 }
+
+class SaveInput {
+  final String userType;
+  final String username;
+  final String query;
+  final String retrievedText;
+  final String modelType;
+  final String reaction;
+
+  SaveInput({
+    required this.userType,
+    required this.username,
+    required this.query,
+    required this.retrievedText,
+    required this.modelType,
+    required this.reaction,
+  });
+
+  Map<String, dynamic> toJson() {
+    return {
+      'user_type': userType,
+      'username': username,
+      'query': query,
+      'retrieved_text': retrievedText,
+      'model_type': modelType,
+      'reaction': reaction,
+    };
+  }
+}</code></pre>
+                </div>
+            </section>
         </main>
     </div>